Find the Kth largest and Kth smallest number in an Array (Python)